package com.czkt;

import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.engine.VelocityTemplateEngine;

public class CodeGenerator {

    public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:mysql://localhost:3316/crm?useUnicode=true&characterEncoding=utf8&serverTimezone=Asia/Shanghai",
                "root", "root")
                .globalConfig(builder -> {
                    builder.author("liu") // 设置作者
//                            .enableSwagger() // 开启 swagger 模式
                            .outputDir("C:\\Users\\10261\\Desktop\\代码生成"); // 指定输出目录

                })
//                .dataSourceConfig(builder ->
//                        builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
//                            int typeCode = metaInfo.getJdbcType().TYPE_CODE;
//                            if (typeCode == Types.SMALLINT) {
//                                // 自定义类型转换
//                                return DbColumnType.INTEGER;
//                            }
//                            return typeRegistry.getColumnType(metaInfo);
//                        })
//                )
                .packageConfig(builder ->
                        builder.parent("com.czkt") // 设置父包名
//                                .moduleName("system") // 设置父包模块名
//                                .pathInfo(Collections.singletonMap(OutputFile.xml, "D://")) // 设置mapperXml生成路径
                )
//                .strategyConfig(
////                        builder ->
////                        builder.addInclude("t_simple") // 设置需要生成的表名
////                                .addTablePrefix("t_", "c_") // 设置过滤表前缀
//                )
                .templateEngine(new VelocityTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
//                .templateConfig(builder -> builder.xml())
                .execute();
    }
}
